hackerrank gradient students problem solution in python